home *** CD-ROM | disk | FTP | other *** search
/ Programming an RTS Game with Direct3D / Programming an RTS Game with Direct3D.iso / Examples / Chapter 9 / Example 9.2 / shader.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  2005-12-08  |  2.3 KB  |  100 lines
  1. #include "shader.h"
  2.  
  3. SHADER::SHADER()
  4. {
  5.     PixelShader = NULL;
  6.     VertexShader = NULL;
  7.     ConstantTable = NULL;
  8. }
  9.  
  10. SHADER::~SHADER()
  11. {
  12.     if(PixelShader != NULL)
  13.     {
  14.         PixelShader->Release();
  15.         PixelShader = NULL;
  16.     }
  17.  
  18.     if(VertexShader != NULL)
  19.     {
  20.         VertexShader->Release();
  21.         VertexShader = NULL;
  22.     }
  23. }
  24.  
  25. void SHADER::Init(IDirect3DDevice9 *Dev, const char fName[], int typ)
  26. {
  27.     Device = Dev;
  28.     type = typ;
  29.     if(Device == NULL)return;
  30.  
  31.     // Assemble and set the pixel or vertex shader     
  32.     HRESULT hRes;
  33.     LPD3DXBUFFER Code = NULL;
  34.     LPD3DXBUFFER ErrorMsgs = NULL;
  35.  
  36.     if(type == PIXEL_SHADER)
  37.         hRes = D3DXCompileShaderFromFile(fName, NULL, NULL, "Main", "ps_2_0", D3DXSHADER_DEBUG, &Code, &ErrorMsgs, &ConstantTable);
  38.     else hRes = D3DXCompileShaderFromFile(fName, NULL, NULL, "Main", "vs_2_0", D3DXSHADER_DEBUG, &Code, &ErrorMsgs, &ConstantTable);
  39.  
  40.     if((FAILED(hRes)) && (ErrorMsgs != NULL))        //If failed
  41.     {
  42.         if(type == PIXEL_SHADER)
  43.             debug.Print("Couldnt compile the Pixel Shader");
  44.         else debug.Print("Couldnt compile the Vertex Shader");
  45.  
  46.         debug.Print((char*)ErrorMsgs->GetBufferPointer());
  47.         return;
  48.     }
  49.  
  50.     if(type == PIXEL_SHADER)
  51.         hRes = Device->CreatePixelShader((DWORD*)Code->GetBufferPointer(), &PixelShader);
  52.     else hRes = Device->CreateVertexShader((DWORD*)Code->GetBufferPointer(), &VertexShader);
  53.  
  54.     if(FAILED(hRes))
  55.     {
  56.         if(type == PIXEL_SHADER)
  57.             debug.Print("Couldnt Create the Pixel Shader");
  58.         else debug.Print("Couldnt Create the Vertex Shader");
  59.         exit(0);
  60.     }
  61. }
  62.  
  63. void SHADER::Begin()
  64. {
  65.     if(type == PIXEL_SHADER)
  66.         HRESULT hRes = Device->SetPixelShader(PixelShader);
  67.     else HRESULT hRes = Device->SetVertexShader(VertexShader);
  68. }
  69.  
  70. void SHADER::End()
  71. {
  72.     if(type == PIXEL_SHADER)
  73.         HRESULT hRes = Device->SetPixelShader(NULL);
  74.     else HRESULT hRes = Device->SetVertexShader(NULL);
  75. }
  76.  
  77. D3DXHANDLE SHADER::GetConstant(char name[])
  78. {
  79.     return ConstantTable->GetConstantByName(NULL, name);
  80. }
  81.  
  82. void SHADER::SetFloat(D3DXHANDLE h, float f)
  83. {
  84.     ConstantTable->SetFloat(Device, h, f);
  85. }
  86.  
  87. void SHADER::SetVector3(D3DXHANDLE h, D3DXVECTOR3 v)
  88. {
  89.     ConstantTable->SetValue(Device, h, &v, sizeof(D3DXVECTOR3));
  90. }
  91.  
  92. void SHADER::SetVector4(D3DXHANDLE h, D3DXVECTOR4 v)
  93. {
  94.     ConstantTable->SetVector(Device, h, &v);
  95. }
  96.  
  97. void SHADER::SetMatrix(D3DXHANDLE h, D3DXMATRIX m)
  98. {
  99.     ConstantTable->SetMatrix(Device, h, &m);
  100. }